#pragma once
#include <iostream>
using namespace std;
#include <string>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <strings.h>
#include <functional>

#define BUFSIZE 1024

enum errnum
{
    SOCK_FAILED = 1,
    BIND_FAILED
};

class udpserver
{
    typedef function<void(int, string, uint16_t, string)> func; // 回调函数类型

public:
    // 构造
    udpserver(uint16_t serverPort, func callback)
        : _sockfd(-1), _serverPort(serverPort), _callback(callback)
    {
    }

    // 析构
    ~udpserver()
    {
    }

    //初始化
    void init()
    {
        // 1、创建套接字
        _sockfd = socket(AF_INET, SOCK_DGRAM, 0);
        if (_sockfd == -1) // 创建失败
        {
            cerr << "套接字创建失败" << endl;
            exit(SOCK_FAILED);
        }
        // 套接字创建成功
        cout << "套接字创建成功 sockfd:" << _sockfd << endl;

        // 2、绑定ip地址和端口号
        struct sockaddr_in serverinfor;
        // 初始化套接字属性
        bzero(&serverinfor, sizeof(struct sockaddr_in));
        serverinfor.sin_family = AF_INET;
        serverinfor.sin_port = htons(_serverPort);
        serverinfor.sin_addr.s_addr = htonl(INADDR_ANY);

        int n = bind(_sockfd, (struct sockaddr *)&serverinfor, sizeof(struct sockaddr_in));
        if (n == -1)
        {
            cerr << "套接字绑定失败" << endl;
            exit(BIND_FAILED);
        }
        // 套接字绑定成功
    }

    // 启动
    void start()
    {
        // 从网络中读取数据
        char buffer[BUFSIZE];                       // 缓冲区，暂存读取到的数据
        struct sockaddr_in clientinfor;             // 用于从消息中提取客户端信息
        socklen_t len = sizeof(struct sockaddr_in); // 属性结构的大小
        bzero(&clientinfor, sizeof(struct sockaddr_in));

        while (true)
        {
            int n = recvfrom(_sockfd, buffer, sizeof(buffer) - 1, 0, (struct sockaddr *)&clientinfor, &len);
            if (n > 0) // 消息读取成功
            {
                buffer[n] = 0; // 在读取到的数据后面添加\0
                string clientIp = inet_ntoa(clientinfor.sin_addr);
                uint16_t clientPort = ntohs(clientinfor.sin_port);
                cout << clientIp << "[" << clientPort << "]# " << buffer << endl;

                _callback(_sockfd, clientIp, clientPort, buffer); // 根据接收到的消息执行对应任务
            }
        }
    }

private:
    int _sockfd;          // 套接字id
    uint16_t _serverPort; // 服务器端口号
    func _callback;       // 回调函数
};